package com.dahe.ysxt.vector.audit.service;

import com.dahe.ysxt.vector.audit.entity.AuditResult;
import com.dahe.ysxt.vector.audit.entity.LikeAuditResult;
import com.tencent.tcvectordb.client.VectorDBClient;

import java.util.List;

public interface AuditService {
    /**
     * 创建集合
     */
    void createCollection();
    /**
     * 将DB中的审计结果转入向量数据库
     * @return
     */
    int batchAuditDB();

    /**
     * 批量插入数据导向量数据库
     * @param client
     * @param auditResultList
     */
    void batchAuditVector(VectorDBClient client, List<AuditResult> auditResultList);

    /**
     * 查询相似性
     * @param id
     * @param errorWords
     * @param correctWords
     * @param conText
     * @return
     */
    LikeAuditResult queryByEmbedding(Long id, String errorWords, String correctWords, String conText);

    void batchSetEmbedding();
}
